package cn.edu.zut.sunday.blimp.domain.constant;

import java.util.HashMap;
import java.util.Map;

public enum TableHeader {
    DATASET_NAME(datasetInitMap()),
    EXAMPLE_INFO(exampleInfoInitMap());

    private final Map<String, String> header;

    private TableHeader(Map<String, String> map) {
        this.header = map;
    }

    public Map<String, String> getHeader() {
        return this.header;
    }

    private static Map<String, String> datasetInitMap() {
        Map<String, String> map = new HashMap<>();
        map.put("id", "ID");
        map.put("datasetName", "数据集文件名");
        map.put("count", "样本数量");
        map.put("md5Info", "md5信息");
        map.put("fileType", "文件类型");
        map.put("createTime", "创建时间");
        map.put("updateTime", "修改时间");

        return map;
    }

    private static Map<String, String> exampleInfoInitMap() {
        Map<String, String> map = new HashMap<>();
        map.put("id", "ID");
        map.put("datasetInfo", "所属数据集名/ID");
        map.put("sentenceGood", "好句子");
        map.put("sentenceBad", "坏句子");
        map.put("field", "语法/语义");
        map.put("linguisticsTerm", "语言学术语");
        map.put("uid", "UID");
        map.put("createTime", "创建时间");
        map.put("updateTime", "修改时间");

        return map;
    }
}
